#!/usr/bin/env python3
import os
import shutil
import sys


def move_files_starting_with_A_or_H(source_dir, target_dir):
    # 确保目标目录存在
    os.makedirs(target_dir, exist_ok=True)

    # 遍历源目录中的所有条目（仅顶层）
    for item in os.listdir(source_dir):
        item_path = os.path.join(source_dir, item)

        # 仅处理文件（跳过子目录）
        if os.path.isfile(item_path):
            if item.startswith(("A", "H")):
                dest_path = os.path.join(target_dir, item)
                print(f"Moving: {item_path} -> {dest_path}")
                shutil.move(item_path, dest_path)

    print("✅ 移动完成！")


if __name__ == "__main__":

    # source = "/mnt/data1/zzy/LNM/Zhujiang-slide/2019-2021CRC/"
    source = "/mnt/data1/zzy/LNM/珠江医院/wsi/"
    target = "/mnt/data1/zzy/LNM/Zhujiang-slide/u1/"
    if not os.path.isdir(source):
        print(f"错误: 源路径 '{source}' 不是一个有效目录。")
        sys.exit(1)

    move_files_starting_with_A_or_H(source, target)
